import torchvision.transforms as transforms
import PIL
from PIL import Image

# img_path = (r'D:\PyCharm\Unet++\flood\images\4.jpg')
# img = PIL.Image.open(img_path)
# trans=transforms.Resize((256, 256))
# img=trans(img)
# img.show()

import torch
print(torch.cuda.is_available())